/**
* Copyright (c) 2010, Attila Magyar
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*     * Redistributions of source code must retain the above copyright
*       notice, this list of conditions and the following disclaimer.
*     * Redistributions in binary form must reproduce the above copyright
*       notice, this list of conditions and the following disclaimer in the
*       documentation and/or other materials provided with the distribution.
*     * Neither the name of the Attila Magyar nor the
*       names of its contributors may be used to endorse or promote products
*       derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER BE LIABLE FOR ANY
* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/

package org.uftp.net.impl;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.InetSocketAddress;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.Random;
import org.uftp.FTPSocketException;
import org.uftp.log.FTPLogManager;
import org.uftp.log.FTPLogger;
import org.uftp.net.IFTPClientSocket;
import org.uftp.net.IFTPServerSocket;

/**
 * The J2SE version of FTPServerSocket
 * @author Zeroflag
 */
public class FTPServerSocket implements IFTPServerSocket {
        
    private static FTPLogger logger = FTPLogManager.getDefaultLogger( FTPServerSocket.class.getName() );
    
    private ServerSocket serverSocket = null;
    private IFTPClientSocket clientSocket = null;
    private String bindHost;
    private int portRangeBegin;
    private int portRangeEnd;
    
    /** Creates a new instance of FTPServerSocket */
    public FTPServerSocket( int portRangeBegin, int portRangeEnd, String bindHost ) throws FTPSocketException {
        
        this.portRangeBegin = portRangeBegin;
        this.portRangeEnd   = portRangeEnd;
        this.bindHost = bindHost;
        
        try {
            serverSocket = new ServerSocket(); 
        } catch (IOException e) {
           throw new FTPSocketException( e );
        }
    }

    public void close() throws FTPSocketException {
        try {                        
            
            if ( clientSocket != null ) {
                logger.debug( "Closing client socket" );
                clientSocket.close();
                clientSocket = null;
            }            
        }
        finally {
            try {
                logger.debug( "Closing server socket" );
                serverSocket.close();
            } catch (IOException ex) {        
                throw new FTPSocketException( ex );
            }
        }
    }

    public InputStream getInputStream() throws FTPSocketException {
        if ( clientSocket == null ) {
            accept();
        }
        return clientSocket.getInputStream();
    }

    public OutputStream getOutputStream() throws FTPSocketException {
        if ( clientSocket == null ) {
            accept();
        }
        return clientSocket.getOutputStream();        
    }

    public int bind() throws FTPSocketException {        
        try {                                    
            int port = portRangeBegin + ( new Random().nextInt() % ( portRangeEnd - portRangeBegin +1 ) );            
            logger.debug( "Random selected port: " + port );
            serverSocket.bind( new InetSocketAddress( bindHost, port ) );  
            return port;
        } catch (Exception e) {
          throw new FTPSocketException( e );
        }
    }

    public IFTPClientSocket accept() throws FTPSocketException {
        try {    
            logger.debug( "Accepting: " + clientSocket );            
            Socket soc = serverSocket.accept();                        
            logger.debug( "Accepted. ");            
            clientSocket = new FTPClientSocket( soc );
            return clientSocket;
        } catch (Exception e) {
            throw new FTPSocketException( e );
        }
    }

    public boolean isConnected() {
        if ( clientSocket != null ) {
            return clientSocket.isConnected();
        }
        return false;
    }

    public void setPortRange( int begin, int end ) {
        this.portRangeBegin = begin;
        this.portRangeEnd   = end;
    }
    
    public int[] getPortRange() {
        return new int[]{ portRangeBegin, portRangeEnd };
    }
}
